# 指定文件编码为UTF-8，支持中文显示
# coding: utf-8

# 导入NumPy数值计算库
import numpy as np

# 导入matplotlib的pylab绘图模块
import matplotlib.pylab as plt

# 定义sigmoid激活函数
def sigmoid(x):
    # 计算sigmoid函数值：1/(1+e^-x)
    return 1 / (1 + np.exp(-x))

# 定义阶跃函数
def step_function(x):
    # 将输入转换为布尔值后转为整数(0或1)
    return np.array(x > 0, dtype=np.int)

# 生成x轴数据：从-5.0到5.0，步长0.1
x = np.arange(-5.0, 5.0, 0.1)

# 计算sigmoid函数的输出
y1 = sigmoid(x)

# 计算阶跃函数的输出
y2 = step_function(x)

# 绘制sigmoid函数曲线(默认实线)
plt.plot(x, y1)

# 绘制阶跃函数曲线(黑色虚线)
plt.plot(x, y2, 'k--')

# 设置y轴显示范围
plt.ylim(-0.1, 1.1) #指定图中绘制的y轴的范围

# 显示图形窗口
plt.show()
